#!/usr/bin/env python
"""
为迁移站点，创建转向链接页面。
为每个已有的站点html文件生成一个页面；
将生成的文件上传到你的旧网站。
"""

import os


def siteForward(
    sitefilesdir, uploaddir, template, servername, homedir
):
    """
    sitefilesdir: 站点文件在本地的路径
    uploaddir: 准备存放转向链接文件的目录
    template: 待生成的页面的模板
    servername: 站点迁移的目的地，如“www.houyuze.xyz/”
    homedir: 站点的根目录，如“blog/”
    """

    # 保证已创建上传目录
    try:
        os.mkdir(uploaddir)
    except OSError:
        pass

    sitefiles = os.listdir(sitefilesdir)
    template = open(template).read()
    count = 0

    for file in sitefiles:
        if file.endswith('.html') or file.endswith('.htm'):
            fwdname = os.path.join(uploaddir, file)
            print('创建', file, '的重定向文件', fwdname, sep='')

            text = template.replace('$server$', servername)
            text = text.replace('$home$', homedir)
            text = text.replace('$file$', file)

            open(fwdname, 'w').write(text)
            count += 1

    print('最后一个重定向文件 ->\n', text, sep='')
    print('完成：创建了', count, '个重定向文件', sep='')
